import os
import json


# 如果密码文件pwd.txt存在
if os.path.exists('pwd.txt'):
    with open('pwd.txt', 'r', encoding='utf-8') as fr:
        # 将pwd.txt的文件内容用json格式读取到pwd字典变量中。
        pwd = json.load(fr)
# 如果密码文件不存在，就初始化一个pwd字典
else:
    pwd = {}

# 输入一个网站的名字
i = input('Please input sitename: ')

# 如果pwd字典中存在这个网站名字
if pwd.get(i):
    # 将字典中此网站相应的密码复制到系统剪贴板中
    os.system('echo {} | clip'.format(pwd.get(i)))
    print('site: {} password has copied'.format(i))
# 如果不存在
else:
    # 输入密码
    p = input('Please input password: ')
    # 将密码保存到字典中
    pwd[i] = p
    # 将该字典以json格式保存到文件中。
    with open('pwd.txt', 'w', encoding='utf-8') as fw:
        json.dump(pwd, fw)
    print('password saved.')
